home *** CD-ROM | disk | FTP | other *** search
/ Clickx 96 / Clickx 96.iso / software / tools / tool / xbmc-10.1.exe / addons / script.module.pysqlite / lib / pysqlite2 / test / py25tests.py < prev    next >
Encoding:
Python Source  |  2009-10-19  |  2.7 KB  |  81 lines
  1. #-*- coding: ISO-8859-1 -*-
  2. # pysqlite2/test/regression.py: pysqlite regression tests
  3. #
  4. # Copyright (C) 2007 Gerhard HΣring <gh@ghaering.de>
  5. #
  6. # This file is part of pysqlite.
  7. #
  8. # This software is provided 'as-is', without any express or implied
  9. # warranty.  In no event will the authors be held liable for any damages
  10. # arising from the use of this software.
  11. #
  12. # Permission is granted to anyone to use this software for any purpose,
  13. # including commercial applications, and to alter it and redistribute it
  14. # freely, subject to the following restrictions:
  15. #
  16. # 1. The origin of this software must not be misrepresented; you must not
  17. #    claim that you wrote the original software. If you use this software
  18. #    in a product, an acknowledgment in the product documentation would be
  19. #    appreciated but is not required.
  20. # 2. Altered source versions must be plainly marked as such, and must not be
  21. #    misrepresented as being the original software.
  22. # 3. This notice may not be removed or altered from any source distribution.
  23.  
  24. from __future__ import with_statement
  25. import unittest
  26. import pysqlite2.dbapi2 as sqlite
  27.  
  28. did_rollback = False
  29.  
  30. class MyConnection(sqlite.Connection):
  31.     def rollback(self):
  32.         global did_rollback
  33.         did_rollback = True
  34.         sqlite.Connection.rollback(self)
  35.  
  36. class ContextTests(unittest.TestCase):
  37.     def setUp(self):
  38.         global did_rollback
  39.         self.con = sqlite.connect(":memory:", factory=MyConnection)
  40.         self.con.execute("create table test(c unique)")
  41.         did_rollback = False
  42.  
  43.     def tearDown(self):
  44.         self.con.close()
  45.  
  46.     def CheckContextManager(self):
  47.         """Can the connection be used as a context manager at all?"""
  48.         with self.con:
  49.             pass
  50.  
  51.     def CheckContextManagerCommit(self):
  52.         """Is a commit called in the context manager?"""
  53.         with self.con:
  54.             self.con.execute("insert into test(c) values ('foo')")
  55.         self.con.rollback()
  56.         count = self.con.execute("select count(*) from test").fetchone()[0]
  57.         self.failUnlessEqual(count, 1)
  58.  
  59.     def CheckContextManagerRollback(self):
  60.         """Is a rollback called in the context manager?"""
  61.         global did_rollback
  62.         self.failUnlessEqual(did_rollback, False)
  63.         try:
  64.             with self.con:
  65.                 self.con.execute("insert into test(c) values (4)")
  66.                 self.con.execute("insert into test(c) values (4)")
  67.         except sqlite.IntegrityError:
  68.             pass
  69.         self.failUnlessEqual(did_rollback, True)
  70.  
  71. def suite():
  72.     ctx_suite = unittest.makeSuite(ContextTests, "Check")
  73.     return unittest.TestSuite((ctx_suite,))
  74.  
  75. def test():
  76.     runner = unittest.TextTestRunner()
  77.     runner.run(suite())
  78.  
  79. if __name__ == "__main__":
  80.     test()
  81.